package tree;

import java.util.ArrayList;
import java.util.List;

/** 
 * 树节点管理类 
 *  
 * @author bearsmall 
 * 
 */  
public class TreeNodeManager {  
    private List<TreeNode> list;// 树的所有节点  
  
    public TreeNodeManager(TreeNode[] items) {  
        list = new ArrayList<TreeNode>();  
        for (TreeNode treeNode : items) {  
            list.add(treeNode);  
        }  
    }  
  
    public TreeNodeManager(List<TreeNode> items) {  
        list = items;  
    }  
  
    /** 
     * 根据节点ID获取一个节点 
     *  
     * @param id 
     *            节点ID 
     * @return 对应的节点对象 
     */  
    public TreeNode getTreeNodeAT(String id)
    {  
        for (TreeNode treeNode : list)
        {  
            if (treeNode.getId().equals(id))
                return treeNode;  
        }  
        return null;  
    }  
  
    /** 
     * 获取树的根节点 
     *  
     * @return 一棵树的根节点 
     */  
    public TreeNode getRoot() {  
        for (TreeNode treeNode : list) {  
            if (treeNode.getPid().equals("0"))  
                return treeNode;  
        }  
        return null;  
    }  
}